# 05-ATM.py
# 读取文件内容
def get_file():
    # 1、打开文件
    with open('./15-money.txt',"r") as file:
    # 2、获取内容
        str = file.read()
        return str
# 查询余额
def get_money(name):
    print('------------------查询余额-----------------')
    print(f'{name},你好，您的余额剩余：{get_file()}')
# 存款
def add_money(name):
    a = int(input('请输入存款金额：'))
    # 局部变量 之前的余额？ 计算最后的金额
    money = int(get_file())+a 
    # 把计算后的金额存储在文本中  写入内容
    with open('./15-money.txt','w') as file:
        file.write(str(money)) # 
        file.flush()
    print(f'{name},你好，你的存款{a}元成功')
    print(f'{name},你好，你的余额剩余{get_file()}元')
# 取款
def update_money(name):
    a = int(input('取款'))
    if a<int(get_file()):
        money = int(get_file())-a 
        with open('./15-money.txt','w') as file:
            file.write(money)
            file.flush()
        print(f'{name},你好，你的取款{a}元成功')
        print(f'{name},你好，你的余额剩余{get_file()}元')
    else:
        print("余额不足")
# 定义主菜单函数
# money = 5000000  # 这里注释掉

def main(name):
    '''
    money:默认银行卡余额
    name:客户的姓名
    '''
    print(f'{name},您好，欢迎来到农商银行ATM,请选择操作:')
    while True:
        print('查询余额','[输入1]')
        print('存款\t','[输入2]')
        print('取款\t','[输入3]')
        print('退出\t','[输入4]')
        index = int(input('请输入您的选择:'))
        if index==4:break
        if index==1:get_money(name)
        if index==2:add_money(name)
        if index==3:update_money(name)
main(input('请输入您的姓名：'))

